#include <iostream>
#include <signal.h>

#include "../comm/httplib.h"
#include "oj_control.hpp"

using namespace httplib;
using namespace ns_control;

static Control *ctrl_p = nullptr;

void Recovery(int signo)
{
    ctrl_p->RecoveryMachines();
}

int main()
{
    signal(SIGQUIT, Recovery);
    // 用户请求的服务路由功能
    Server svr;
    Control ctrl;
    ctrl_p = &ctrl;

    svr.Get("/hello", [](const Request &req, Response &res) {
        res.set_content("Hello World!", "text/plain;charset=utf-8");
    });
    
    // 获取所有的题目列表
    svr.Get("/all_questions", [&ctrl](const Request &req, Response &res) {
        // 返回一张包含所有题目的html网页
        std::string html;
        ctrl.GetAllQuestions(&html);
        res.set_content(html, "text/html;charset=utf-8");
    });

    // 用户根据题目编号，获取题目详情
    // /question/100 -> 正则匹配，\d表示匹配数字，+表示匹配一个或多个
    // R"()"，保持原始字符串内容的原貌，不用做相关的转义
    svr.Get(R"(/question/(\d+))", [&ctrl](const Request &req, Response &res) {
        std::string number = req.matches[1];    // 获取正则匹配的内容
        std::string html;
        // 根据题目编号，获取题目详情
        ctrl.OneQuestion(number, &html);
        res.set_content(html, "text/html;charset=utf-8");
    });

    // 用户提交代码，使用判题功能（1. 每道题的测试用例 2. compile_and_run）
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request &req, Response &res) {
        std::string number = req.matches[1];    // 获取正则匹配的内容
        std::string result_json;
        ctrl.Judge(number, req.body, &result_json);
        res.set_content(result_json, "application/json;charset=utf-8");
    });

    svr.set_base_dir("./wwwroot"); // 网站首页
    svr.listen("0.0.0.0", 8080);
    return 0;
}